#save_screenshot()  方法来截取当前窗口
from  selenium import webdriver
driver = webdriver.Chrome()
driver.get("http://www.baidu.com")
#截取当前窗口，指定截图图片保存位置
driver.save_screenshot("./files/baidu_img.png")

driver.quit()
